﻿using System;
using System.Diagnostics;
using System.Globalization;
using System.Runtime.InteropServices;
using System.ComponentModel.Design;
using Microsoft.Win32;
using Microsoft.VisualStudio;
using Microsoft.VisualStudio.Shell.Interop;
using Microsoft.VisualStudio.OLE.Interop;
using Microsoft.VisualStudio.Shell;
using System.Collections.Generic;
using System.Linq;

namespace SanjaySingh.DefaultBrowserPackage
{
    /// <summary>
    /// This is the class that implements the package exposed by this assembly.
    ///
    /// The minimum requirement for a class to be considered a valid package for Visual Studio
    /// is to implement the IVsPackage interface and register itself with the shell.
    /// This package uses the helper classes defined inside the Managed Package Framework (MPF)
    /// to do it: it derives from the Package class that provides the implementation of the 
    /// IVsPackage interface and uses the registration attributes defined in the framework to 
    /// register itself and its components with the shell.
    /// </summary>
    // This attribute tells the PkgDef creation utility (CreatePkgDef.exe) that this class is
    // a package.
    [PackageRegistration(UseManagedResourcesOnly = true)]
    // This attribute is used to register the informations needed to show the this package
    // in the Help/About dialog of Visual Studio.
    [InstalledProductRegistration("#110", "#112", "1.0", IconResourceID = 400)]
    // This attribute is needed to let the shell know that this package exposes some menus.
    [ProvideMenuResource("Menus.ctmenu", 1)]
    [Guid(GuidList.guidDefaultBrowserPackagePkgString)]
    public sealed class DefaultBrowserPackagePackage : Package
    {

        private Dictionary<MenuCommand, Browser> commandBrowsersAssociations = new Dictionary<MenuCommand, Browser>();

        /// <summary>
        /// Default constructor of the package.
        /// Inside this method you can place any initialization code that does not require 
        /// any Visual Studio service because at this point the package object is created but 
        /// not sited yet inside Visual Studio environment. The place to do all the other 
        /// initialization is the Initialize method.
        /// </summary>
        public DefaultBrowserPackagePackage()
        {
            
        }

        #region Package Members

        /// <summary>
        /// Initialization of the package; this method is called right after the package is sited, so this is the place
        /// where you can put all the initilaization code that rely on services provided by VisualStudio.
        /// </summary>
        protected override void Initialize()
        {
            try
            {
                base.Initialize();

                List<Browser> installedBrowsers = Browser.GetAllInstalledBrowsers().ToList();
                
                OleMenuCommandService mcs = GetService(typeof(IMenuCommandService)) as OleMenuCommandService;

                if (mcs != null)
                {
                    foreach (UiCommand menuCommandValue in Enum.GetValues(typeof(UiCommand)))
                    {
                        if (menuCommandValue > 0)
                        {
                            CommandID menuCommandID = new CommandID(GuidList.guidDefaultBrowserPackageCmdSet, (int) menuCommandValue);
                            MenuCommand menuItem = new MenuCommand(MenuItemCallback, menuCommandID);
                            mcs.AddCommand(menuItem);

                            Browser correspondingBrowser = installedBrowsers.FirstOrDefault(b => b.GetUiCommand() == menuCommandValue);
                            if (correspondingBrowser != null)
                            {
                                this.commandBrowsersAssociations.Add(menuItem, correspondingBrowser);
                            }
                        }
                    }
                }
                UpdateDefaultBrowserUI();
                
            }
            catch
            {
                // ignoreany exceptions,dont let it pass to VS IDE
            }
        }
        #endregion

        /// <summary>
        /// This function is the callback used to execute a command when the a menu item is clicked.
        /// See the Initialize method to see how the menu item is associated to this function using
        /// the OleMenuCommandService service and the MenuCommand class.
        /// </summary>
        private void MenuItemCallback(object sender, EventArgs e)
        {
            try
            {
                MenuCommand menuCommand = sender as MenuCommand;
                
                if (this.commandBrowsersAssociations.ContainsKey(menuCommand))
                {
                    this.commandBrowsersAssociations[menuCommand].MakeDefaultForVisualStudio();
                    UpdateDefaultBrowserUI();
                }
            }
            catch
            {
                // ignoreany exceptions,dont let it pass to VS IDE
            }
        }

        private void UpdateDefaultBrowserUI()
        {
            foreach (var commandBrowserAssociation in this.commandBrowsersAssociations)
            {
                bool isDefault = commandBrowserAssociation.Value.IsDefault;
                if (commandBrowserAssociation.Key.Checked != isDefault)
                {
                    commandBrowserAssociation.Key.Checked = isDefault;
                }
                
            }
        }
    }
}
